program to find armstrong number in python